<?php

// +----------------------------------------------------------------------

// | TOPThink [ WE CAN DO IT JUST THINK ]

// +----------------------------------------------------------------------

// | Copyright (c) 2010 http://topthink.com All rights reserved.

// +----------------------------------------------------------------------

// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )

// +----------------------------------------------------------------------

// | Author: 麦当苗儿 <zuojiazi@vip.qq.com> <http://www.zjzit.cn>

// +----------------------------------------------------------------------

// | ImageImagick.class.php 2013-03-06

// +----------------------------------------------------------------------

namespace Think\Image\Driver;

use Think\Image;

class Imagick{

    /**

     * 图像资源对象

     * @var resource

     */

    private $img;



    /**

     * 图像信息，包括width,height,type,mime,size

     * @var array

     */

    private $info;



    /**

     * 构造方法，可用于打开一张图像

     * @param string $imgname 图像路径

     */

    public function __construct($imgname = null) {

        $imgname && $this->open($imgname);

    }



    /**

     * 打开一张图像

     * @param  string $imgname 图像路径

     */

    public function open($imgname){

        //检测图像文件

        if(!is_file($imgname)) E('不存在的图像文件');



        //销毁已存在的图像

        empty($this->img) || $this->img->destroy();



        //载入图像

        $this->img = new \Imagick(realpath($imgname));



        //设置图像信息

        $this->info = array(

            'width'  => $this->img->getImageWidth(),

            'height' => $this->img->getImageHeight(),

            'type'   => strtolower($this->img->getImageFormat()),

            'mime'   => $this->img->getImageMimeType(),

        );

    }



    /**

     * 保存图像

     * @param  string  $imgname   图像保存名称

     * @param  string  $type      图像类型

     * @param  boolean $interlace 是否对JPEG类型图像设置隔行扫描

     */

    public function save($imgname, $type = null, $interlace = true){

        if(empty($this->img)) E('没有可以被保存的图像资源');



        //设置图片类型

        if(is_null($type)){

            $type = $this->info['type'];

        } else {

            $type = strtolower($type);

            $this->img->setImageFormat($type);

        }



        //JPEG图像设置隔行扫描

        if('jpeg' == $type || 'jpg' == $type){

            $this->img->setImageInterlaceScheme(1);

        }



        //去除图像配置信息

        $this->img->stripImage();



        //保存图像

        $imgname = realpath(dirname($imgname)) . '/' . basename($imgname); //强制绝对路径

        if ('gif' == $type) {

            $this->img->writeImages($imgname, true);

        } else {

            $this->img->writeImage($imgname);

        }

    }



    /**

     * 返回图像宽度

     * @return integer 图像宽度

     */

    public function width(){

        if(empty($this->img)) E('没有指定图像资源');

        return $this->info['width'];

    }



    /**

     * 返回图像高度

     * @return integer 图像高度

     */

    public function height(){

        if(empty($this->img)) E('没有指定图像资源');

        return $this->info['height'];

    }



    /**

     * 返回图像类型

     * @return string 图像类型

     */

    public function type(){

        if(empty($this->img)) E('没有指定图像资源');

        return $this->info['type'];

    }



    /**

     * 返回图像MIME类型

     * @return string 图像MIME类型

     */

    public function mime(){

        if(empty($this->img)) E('没有指定图像资源');

        return $this->info['mime'];

    }



    /**

     * 返回图像尺寸数组 0 - 图像宽度，1 - 图像高度

     * @return array 图像尺寸

     */

    public function size(){

        if(empty($this->img)) E('没有指定图像资源');

        return array($this->info['width'], $this->info['height']);

    }



    /**

     * 裁剪图像

     * @param  integer $w      裁剪区域宽度

     * @param  integer $h      裁剪区域高度

     * @param  integer $x      裁剪区域x坐标

     * @param  integer $y      裁剪区域y坐标

     * @param  integer $width  图像保存宽度

     * @param  integer $height 图像保存高度

     */

    public function crop($w, $h, $x = 0, $y = 0, $width = null, $height = null){

        if(empty($this->img)) E('没有可以被裁剪的图像资源');



        //设置保存尺寸

        empty($width)  && $width  = $w;

        empty($height) && $height = $h;



        //裁剪图片

        if('gif' == $this->info['type']){

            $img = $this->img->coalesceImages();

            $this->img->destroy(); //销毁原图



            //循环裁剪每一帧

            do {

                $this->_crop($w, $h, $x, $y, $width, $height, $img);

            } while ($img->nextImage());

            

            //压缩图片

            $this->img = $img->deconstructImages();

            $img->destroy(); //销毁零时图片

        } else {

            $this->_crop($w, $h, $x, $y, $width, $height);

        }

    }



    /* 裁剪图片，内部调用 */

    private function _crop($w, $h, $x, $y, $width, $height, $img = null){

        is_null($img) && $img = $this->img;



        //裁剪

        $info = $this->info;

        if($x != 0 || $y != 0 || $w != $info['width'] || $h != $info['height']){

            $img->cropImage($w, $h, $x, $y);

            $img->setImagePage($w, $h, 0, 0); //调整画布和图片一致

        }

        

        //调整大小

        if($w != $width || $h != $height){

            $img->sampleImage($width, $height);

        }



        //设置缓存尺寸

        $this->info['width']  = $w;

        $this->info['height'] = $h;

    }



    /**

     * 生成缩略图

     * @param  integer $width  缩略图最大宽度

     * @param  integer $height 缩略图最大高度

     * @param  integer $type   缩略图裁剪类型

     */

    public function thumb($width, $height, $type = Image::IMAGE_THUMB_SCALE){

        if(empty($this->img)) E('没有可以被缩略的图像资源');



        //原图宽度和高度

        $w = $this->info['width'];

        $h = $this->info['height'];



        /* 计算缩略图生成的必要参数 */

        switch ($type) {

            /* 等比例缩放 */

            case Image::IMAGE_THUMB_SCALE:

                //原图尺寸小于缩略图尺寸则不进行缩略

                if($w < $width && $h < $height) return;



                //计算缩放比例

                $scale = min($width/$w, $height/$h);

                

                //设置缩略图的坐标及宽度和高度

                $x = $y = 0;

                $width  = $w * $scale;

                $height = $h * $scale;

                break;



            /* 居中裁剪 */

            case Image::IMAGE_THUMB_CENTER:

                //计算缩放比例

                $scale = max($width/$w, $height/$h);



                //设置缩略图的坐标及宽度和高度

                $w = $width/$scale;

                $h = $height/$scale;

                $x = ($this->info['width'] - $w)/2;

                $y = ($this->info['height'] - $h)/2;

                break;



            /* 左上角裁剪 */

            case Image::IMAGE_THUMB_NORTHWEST:

                //计算缩放比例

                $scale = max($width/$w, $height/$h);



                //设置缩略图的坐标及宽度和高度

                $x = $y = 0;

                $w = $width/$scale;

                $h = $height/$scale;

                break;



            /* 右下角裁剪 */

            case Image::IMAGE_THUMB_SOUTHEAST:

                //计算缩放比例

                $scale = max($width/$w, $height/$h);



                //设置缩略图的坐标及宽度和高度

                $w = $width/$scale;

                $h = $height/$scale;

                $x = $this->info['width'] - $w;

                $y = $this->info['height'] - $h;

                break;



            /* 填充 */

            case Image::IMAGE_THUMB_FILLED:

                //计算缩放比例

                if($w < $width && $h < $height){

                    $scale = 1;

                } else {

                    $scale = min($width/$w, $height/$h);

                }



                //设置缩略图的坐标及宽度和高度

                $neww = $w * $scale;

                $newh = $h * $scale;

                $posx = ($width  - $w * $scale)/2;

                $posy = ($height - $h * $scale)/2;



                //创建一张新图像

                $newimg = new Imagick();

                $newimg->newImage($width, $height, 'white', $this->info['type']);





                if('gif' == $this->info['type']){

                    $imgs = $this->img->coalesceImages();

                    $img  = new Imagick();

                    $this->img->destroy(); //销毁原图



                    //循环填充每一帧

                    do {

                        //填充图像

                        $image = $this->_fill($newimg, $posx, $posy, $neww, $newh, $imgs);

                        

                        $img->addImage($image);

                        $img->setImageDelay($imgs->getImageDelay());

                        $img->setImagePage($width, $height, 0, 0);



                        $image->destroy(); //销毁零时图片



                    } while ($imgs->nextImage());



                    //压缩图片

                    $this->img->destroy();

                    $this->img = $img->deconstructImages();

                    $imgs->destroy(); //销毁零时图片

                    $img->destroy(); //销毁零时图片



                } else {

                    //填充图像

                    $img = $this->_fill($newimg, $posx, $posy, $neww, $newh);

                    //销毁原图

                    $this->img->destroy();

                    $this->img = $img;

                }



                //设置新图像属性

                $this->info['width']  = $width;

                $this->info['height'] = $height;

                return;



            /* 固定 */

            case Image::IMAGE_THUMB_FIXED:

                $x = $y = 0;

                break;



            default:

                E('不支持的缩略图裁剪类型');

        }



        /* 裁剪图像 */

        $this->crop($w, $h, $x, $y, $width, $height);

    }



    /* 填充指定图像，内部使用 */

    private function _fill($newimg, $posx, $posy, $neww, $newh, $img = null){

        is_null($img) && $img = $this->img;



        /* 将指定图片绘入空白图片 */

        $draw  = new ImagickDraw();

        $draw->composite($img->getImageCompose(), $posx, $posy, $neww, $newh, $img);

        $image = $newimg->clone();

        $image->drawImage($draw);

        $draw->destroy();



        return $image;

    }



    /**

     * 添加水印

     * @param  string  $source 水印图片路径

     * @param  integer $locate 水印位置

     * @param  integer $alpha  水印透明度

     */

    public function water($source, $locate = Image::IMAGE_WATER_SOUTHEAST,$alpha=80){

        //资源检测

        if(empty($this->img)) E('没有可以被添加水印的图像资源');

        if(!is_file($source)) E('水印图像不存在');



        //创建水印图像资源

        $water = new Imagick(realpath($source));

        $info  = array($water->getImageWidth(), $water->getImageHeight());



        /* 设定水印位置 */

        switch ($locate) {

            /* 右下角水印 */

            case Image::IMAGE_WATER_SOUTHEAST:

                $x = $this->info['width'] - $info[0];

                $y = $this->info['height'] - $info[1];

                break;



            /* 左下角水印 */

            case Image::IMAGE_WATER_SOUTHWEST:

                $x = 0;

                $y = $this->info['height'] - $info[1];

                break;



            /* 左上角水印 */

            case Image::IMAGE_WATER_NORTHWEST:

                $x = $y = 0;

                break;



            /* 右上角水印 */

            case Image::IMAGE_WATER_NORTHEAST:

                $x = $this->info['width'] - $info[0];

                $y = 0;

                break;



            /* 居中水印 */

            case Image::IMAGE_WATER_CENTER:

                $x = ($this->info['width'] - $info[0])/2;

                $y = ($this->info['height'] - $info[1])/2;

                break;



            /* 下居中水印 */

            case Image::IMAGE_WATER_SOUTH:

                $x = ($this->info['width'] - $info[0])/2;

                $y = $this->info['height'] - $info[1];

                break;



            /* 右居中水印 */

            case Image::IMAGE_WATER_EAST:

                $x = $this->info['width'] - $info[0];

                $y = ($this->info['height'] - $info[1])/2;

                break;



            /* 上居中水印 */

            case Image::IMAGE_WATER_NORTH:

                $x = ($this->info['width'] - $info[0])/2;

                $y = 0;

                break;



            /* 左居中水印 */

            case Image::IMAGE_WATER_WEST:

                $x = 0;

                $y = ($this->info['height'] - $info[1])/2;

                break;



            default:

                /* 自定义水印坐标 */

                if(is_array($locate)){

                    list($x, $y) = $locate;

                } else {

                    E('不支持的水印位置类型');

                }

        }



        //创建绘图资源

        $draw = new ImagickDraw();

        $draw->composite($water->getImageCompose(), $x, $y, $info[0], $info[1], $water);

        

        if('gif' == $this->info['type']){

            $img = $this->img->coalesceImages();

            $this->img->destroy(); //销毁原图



            do{

                //添加水印

                $img->drawImage($draw);

            } while ($img->nextImage());



            //压缩图片

            $this->img = $img->deconstructImages();

            $img->destroy(); //销毁零时图片



        } else {

            //添加水印

            $this->img->drawImage($draw);

        }



        //销毁水印资源

        $draw->destroy();

        $water->destroy();

    }



    /**

     * 图像添加文字

     * @param  string  $text   添加的文字

     * @param  string  $font   字体路径

     * @param  integer $size   字号

     * @param  string  $color  文字颜色

     * @param  integer $locate 文字写入位置

     * @param  integer $offset 文字相对当前位置的偏移量

     * @param  integer $angle  文字倾斜角度

     */

    public function text($text, $font, $size, $color = '#00000000', 

        $locate = Image::IMAGE_WATER_SOUTHEAST, $offset = 0, $angle = 0){

        //资源检测

        if(empty($this->img)) E('没有可以被写入文字的图像资源');

        if(!is_file($font)) E("不存在的字体文件：{$font}");



        //获取颜色和透明度

        if(is_array($color)){

            $color = array_map('dechex', $color);

            foreach ($color as &$value) {

                $value = str_pad($value, 2, '0', STR_PAD_LEFT);

            }

            $color = '#' . implode('', $color);

        } elseif(!is_string($color) || 0 !== strpos($color, '#')) {

            E('错误的颜色值');

        }

        $col = substr($color, 0, 7);

        $alp = strlen($color) == 9 ? substr($color, -2) : 0;

        



        //获取文字信息

        $draw = new ImagickDraw();

        $draw->setFont(realpath($font));

        $draw->setFontSize($size);

        $draw->setFillColor($col);

        $draw->setFillAlpha(1-hexdec($alp)/127);

        $draw->setTextAntialias(true);

        $draw->setStrokeAntialias(true);

        

        $metrics = $this->img->queryFontMetrics($draw, $text);



        /* 计算文字初始坐标和尺寸 */

        $x = 0;

        $y = $metrics['ascender'];

        $w = $metrics['textWidth'];

        $h = $metrics['textHeight'];



        /* 设定文字位置 */

        switch ($locate) {

            /* 右下角文字 */

            case Image::IMAGE_WATER_SOUTHEAST:

                $x += $this->info['width']  - $w;

                $y += $this->info['height'] - $h;

                break;



            /* 左下角文字 */

            case Image::IMAGE_WATER_SOUTHWEST:

                $y += $this->info['height'] - $h;

                break;



            /* 左上角文字 */

            case Image::IMAGE_WATER_NORTHWEST:

                // 起始坐标即为左上角坐标，无需调整

                break;



            /* 右上角文字 */

            case Image::IMAGE_WATER_NORTHEAST:

                $x += $this->info['width'] - $w;

                break;



            /* 居中文字 */

            case Image::IMAGE_WATER_CENTER:

                $x += ($this->info['width']  - $w)/2;

                $y += ($this->info['height'] - $h)/2;

                break;



            /* 下居中文字 */

            case Image::IMAGE_WATER_SOUTH:

                $x += ($this->info['width'] - $w)/2;

                $y += $this->info['height'] - $h;

                break;



            /* 右居中文字 */

            case Image::IMAGE_WATER_EAST:

                $x += $this->info['width'] - $w;

                $y += ($this->info['height'] - $h)/2;

                break;



            /* 上居中文字 */

            case Image::IMAGE_WATER_NORTH:

                $x += ($this->info['width'] - $w)/2;

                break;



            /* 左居中文字 */

            case Image::IMAGE_WATER_WEST:

                $y += ($this->info['height'] - $h)/2;

                break;



            default:

                /* 自定义文字坐标 */

                if(is_array($locate)){

                    list($posx, $posy) = $locate;

                    $x += $posx;

                    $y += $posy;

                } else {

                    E('不支持的文字位置类型');

                }

        }



        /* 设置偏移量 */

        if(is_array($offset)){

            $offset = array_map('intval', $offset);

            list($ox, $oy) = $offset;

        } else{

            $offset = intval($offset);

            $ox = $oy = $offset;

        }



        /* 写入文字 */

        if('gif' == $this->info['type']){

            $img = $this->img->coalesceImages();

            $this->img->destroy(); //销毁原图

            do{

                $img->annotateImage($draw, $x + $ox, $y + $oy, $angle, $text);

            } while ($img->nextImage());



            //压缩图片

            $this->img = $img->deconstructImages();

            $img->destroy(); //销毁零时图片



        } else {

            $this->img->annotateImage($draw, $x + $ox, $y + $oy, $angle, $text);

        }

        $draw->destroy();

    }



    /**

     * 析构方法，用于销毁图像资源

     */

    public function __destruct() {

        empty($this->img) || $this->img->destroy();

    }

}